import qgis
from qgis.core import *
from qgis.analysis import *
from os.path import join
from PyQt4.QtCore import *
from osgeo import gdal, ogr,osr
import math
from datetime import datetime, date, time, timedelta


def v_copia(path_vector,path_salida):
    """
    Crea una copia de la capa a partir de la ruta de la capa, la capa es creada con el mismo sistema de
    referencia que el origen.
    """
    vlayer = QgsVectorLayer(path_vector,"","ogr")
    clonarv = QgsVectorFileWriter.writeAsVectorFormat(vlayer,path_salida,'utf-8',vlayer.crs(),"ESRI Shapefile")



def crear_campo(path_vector,nombre_campo,tipo):

    ''' Crear_campo(path_vector,nombre_campo,tipo)
    Esta funcion crea un campo segun el tipo especificado.
    path_vector:La ruta del archivo shapefile al cual se le quiere agregar el campo
    nombre_campo: Nombre del campo nuevo
    tipo: es el tipo de campo que se quiere crear:
    Int: para crear un campo tipo entero
    Double: para crear un campo tipo doble o flotante
    String: para crear un campo tipo texto
    Date: para crear un campo tipo fecha '''

    if len(nombre_campo) >10:
        print("el nombre del campo debe contener maximo 10 caracteres")
    else:
        if tipo == "Int":
            nombre=QgsVectorLayer(path_vector,"","ogr")
            nombre.dataProvider().addAttributes([QgsField(nombre_campo.lower(),QVariant.Int)])
            return nombre
        elif tipo == "Double":
            nombre=QgsVectorLayer(path_vector,"","ogr")
            nombre.dataProvider().addAttributes([QgsField(nombre_campo.lower(),QVariant.Double)])
            return nombre
        elif tipo == "String":
            nombre=QgsVectorLayer(path_vector,"","ogr")
            nombre.dataProvider().addAttributes([QgsField(nombre_campo.lower(),QVariant.String)])
            return nombre
        elif tipo == "Date":
            nombre=QgsVectorLayer(path_vector,"","ogr")
            nombre.dataProvider().addAttributes([QgsField(nombre_campo.lower(),QVariant.Date)])
            return nombre
        else:
            print ("el tipo no existe o hay error en su declaracion")

sig_path="C:/Dropbox (LANCIS)/SIG/desarrollo/sig_megadapt/procesamiento/prec_volumen/"
precipitacion = QgsVectorLayer (sig_path+"insumos/ageb_prec_00_17_utm.shp","","ogr")

v_copia(sig_path+"insumos/ageb_prec_00_17_utm.shp"  , sig_path+"salida/ageb_vol_prec_utm.shp")

vol_prec = QgsVectorLayer (sig_path+"salida/ageb_vol_prec_utm.shp","","ogr")

campos=[field.name() for field in precipitacion.fields()] [1:]

for campo in campos:
    if campo !='area':
        print campo
        campo_v=campo+'v'
        crear_campo(sig_path+"salida/ageb_vol_prec_utm.shp",campo_v,"Double")
        
        vol_prec.startEditing()
        for ageb in vol_prec.getFeatures():
            volumen = (ageb[campo]*ageb['area'])/1000
            ageb[campo_v]=volumen
            vol_prec.updateFeature(ageb)
            #print "el ageb ",ageb['ageb_id'],"para el anio",campo, "tiene un volumen de " , volumen, "dato del shape",ageb[campo]
        vol_prec.commitChanges()
        
print "el proceso se termino"